from die import Die
import pygal

# 创建一个6面骰子和一个10面骰子
die_1 = Die()
die_2 = Die(10)

# 随机投掷几次骰子，将结果存在一个列表中
results=[]
for roll_num in range(50000):
    result = die_1.roll() + die_2.roll()
    results.append(result)

# print(results)

# 分析结果
frequencies = []
max_result = die_1.num_sides + die_2.num_sides
for value in range(2,max_result+1):
    frequency = results.count(value)
    frequencies.append(frequency)

print(frequencies)

# 绘制直方图
hist = pygal.Bar()
  
hist.title = "Results of rolling a D6 and a D10 50000 times"
# hist.x_labels = ['2','3','4','5','6','7','8','9','10','11','12','13','14','15','16']
hist.x_labels = [x for x in range(2,die_1.num_sides+die_2.num_sides+1)]

hist.x_title = "Result"
hist.y_title = "Frequency of Result"

hist.add("D6 + D10",frequencies)
# 最后生成的图标图片格式必须为svg
hist.render_to_file("different_dice_visual.svg")